package com.example.excel;

import com.alibaba.excel.EasyExcel;

import java.io.File;
import java.io.IOException;
import java.util.List;

public class ExcelUtil {

/*    public static void createExcel(List user) {
        Long start = System.currentTimeMillis();
        //1、创建一个文件对象
        File excelFile = new File("C:\\Users\\admin\\Desktop\\test.xlsx");
        //2、判断文件是否存在，不存在则创建一个Excel文件
        if (!excelFile.exists()) {
            try {
                excelFile.createNewFile();//创建一个新的文件
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
        //3、指定需要那个class去写。然后写到第一个sheet，名字为模版，然后文件流会自动关闭
        EasyExcel.write(excelFile, UserDto.class).sheet("模版").doWrite(user);
        Long end = System.currentTimeMillis();
        System.out.println((end - start) + "ms");
    }*/

    public static void createExcel(ExcelInfo excelInfo) {
        Long start = System.currentTimeMillis();
        //1、创建一个文件对象
        File excelFile = new File(excelInfo.getAddress());
        //2、判断文件是否存在，不存在则创建一个Excel文件
        if (!excelFile.exists()) {
            try {
                excelFile.createNewFile();//创建一个新的文件
            } catch (IOException e) {
                e.printStackTrace();
            }
        }
        //3、指定需要那个class去写。然后写到第一个sheet，名字为模版，然后文件流会自动关闭
        EasyExcel.write(excelFile, UserDto.class).sheet("1").doWrite(excelInfo.getUserList());
        Long end = System.currentTimeMillis();
        System.out.println((end - start) + "ms");
    }

}
